how tf.one_hot function works